/*
 * Copyright 2017 Red Hat, Inc. and/or its affiliates.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.jbpm.persistence;


import javax.persistence.EntityManager;

import org.drools.persistence.jpa.JpaPersistenceContextManager;
import org.jbpm.persistence.api.ProcessPersistenceContext;
import org.jbpm.persistence.api.ProcessPersistenceContextManager;
import org.kie.api.runtime.Environment;
import org.kie.api.runtime.EnvironmentName;

public class JpaProcessPersistenceContextManager extends JpaPersistenceContextManager
    implements
    ProcessPersistenceContextManager {

    public JpaProcessPersistenceContextManager(Environment env) {
        super( env );
    }

    public ProcessPersistenceContext getProcessPersistenceContext() {
        Boolean locking = (Boolean) env.get(EnvironmentName.USE_PESSIMISTIC_LOCKING);
        if( locking == null ) { 
            locking = false;
        }
        String lockingMode = (String) env.get(EnvironmentName.USE_PESSIMISTIC_LOCKING_MODE);
        
        boolean useJTA = true; 
        return new JpaProcessPersistenceContext( getCommandScopedEntityManager(), useJTA, locking, lockingMode, txm );
    }

    @Override
    public EntityManager getCommandScopedEntityManager() {
        EntityManager em = super.getCommandScopedEntityManager();
        // ensure em is set in the environment to cover situation when em is taken from tx directly
        // when using per process instance runtime strategy
        if (env.get(EnvironmentName.CMD_SCOPED_ENTITY_MANAGER) == null) {
            env.set(EnvironmentName.CMD_SCOPED_ENTITY_MANAGER, em);
        }
        
        return em;
    }

}
